cd ../..

gpus=$1
psd_lo_thd=0.10
psd_hi_thd=0.90
weight=175
mask_len=4
mask_ratio=0.4
lr_scale=0.5
wMT=70
for bsz in 24 #  
do
    python -m trainer.ATSTFTContrastiveTrainer fit \
        --config configs/model/ATSTSED_contrastive.yaml \
        --config configs/dataset/desed_with_id.yaml \
        --config configs/trainer/default.yaml \
        --config configs/experiment/intplcontrastive.yaml \
        --config configs/experiment/plugin_ATSTFrame_finetune.yaml \
        --model.exp_name="5_cl_usage/ATSTSED/stage2/+wCL_${weight}_wMT_${wMT}/0.1_scheduler/unlabeled_bsz${bsz}/lr_scale${lr_scale}/mask_len${mask_len}/mask_ratio${mask_ratio}" \
        --data.init_args.batch_sizes=[12,12,24,${bsz}] \
        --model.cl_decay=True \
        --model.cl_weight=${weight} \
        --trainer.devices=${gpus} \
        --trainer.strategy="ddp_find_unused_parameters_false" \
        --model.psd_label_filter=False \
        --model.psd_lo_thd=${psd_lo_thd} \
        --model.psd_hi_thd=${psd_hi_thd} \
        --model.sed_model.init_args.atst_train=True \
        --model.unsupervised_loss.init_args.weight=${wMT} \
        --model.val_psds=False \
        --trainer.check_val_every_n_epoch=10 \
        --model.contrastive_weight.init_args.min_weight=0.31623 \
        --model.ckpt_path="/home/shaonian/sed/logs/5_cl_usage/ATSTSED/stage1/+wCL_5_wMT_2/0.1_scheduler/unlabeled_bsz24/baseline/checkpoints/epoch189_metric0.1048.ckpt" \
        --data.init_args.contrastive_aug.init_args.mask_ratio=${mask_ratio} \
        --model.lr_scale=${lr_scale} \
        --data.init_args.contrastive_aug.init_args.mask_len=${mask_len}
    done